package com.ccr.common.utils;

import com.ccr.common.exception.ExcelException;
import jxl.Cell;
import jxl.Sheet;
import jxl.Workbook;
import jxl.write.Label;
import jxl.write.WritableSheet;
import jxl.write.WritableWorkbook;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang.time.DateFormatUtils;
import org.apache.commons.lang.time.DateUtils;

import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.lang.reflect.Field;
import java.util.*;

/**
 * Created by Chengrui on 2015/6/24.
 * @Description Excel导入导出工具类
 */
public class ExcelUtils {

    /**
     * @MethodName  : listToExcel
     * @Description : 导出Excel（可以导出到本地文件系统，也可以导出到浏览器，可自定义工作表大小）
     * @param list      数据源
     * @param fieldMap  类的英文属性和Excel中的中文列名的对应关系 例：{id=编号}
     * 如果需要的是引用对象的属性，则英文属性使用类似于EL表达式的格式
     * 如：list中存放的都是student，student中又有college属性，而我们需要学院名称，则可以这样写
     * fieldMap.put("college.collegeName","学院名称")
     * @param sheetName 工作表的名称
     * @param sheetSize 每个工作表中记录的最大个数
     * @param os       导出流
     * @throws ExcelException
     */
    public static <T> void listToExcel(
            List<T> list,
            LinkedHashMap<String, String> fieldMap,
            String sheetName,
            int sheetSize,
            OutputStream os
    ) throws ExcelException {

        if(CollectionUtils.isEmpty(list)){
            throw new ExcelException("数据源中没有任何数据");
        }

        if (sheetSize < 1 || sheetSize > 65535){
            sheetSize = 65535;
        }

        WritableWorkbook wwb;
        try {
            //创建工作簿并发送到OutputStream指定的地方
            wwb = Workbook.createWorkbook(os);

            //因为2003的Excel一个工作表最多可以有65536条记录，除去列头剩下65535条
            //所以如果记录太多，需要放到多个工作表中，其实就是个分页的过程
            //1.计算一共有多少个工作表
            double sheetNum = Math.ceil(list.size()/new Integer(sheetSize).doubleValue());
            //2.创建相应的工作表,并向其中填充数据
            for (int i=0;i<sheetNum;i++){
                //如果只有一个工作表的情况
                if (sheetNum == 1){
                    WritableSheet sheet = wwb.createSheet(sheetName, i);
                    //向工作表中填充数据
                    fillSheet(sheet, list, fieldMap, 0, list.size()-1);

                //有多个工作表的情况
                } else {
                    WritableSheet sheet = wwb.createSheet(sheetName+(i+1),i);
                    //获取开始索引和结束索引
                    int firstIndex = i*sheetSize;
                    int lastIndex = (i+1)*sheetSize-1>list.size()-1 ? list.size()-1 : (i+1)*sheetSize-1;
                    //填充工作表
                    fillSheet(sheet, list, fieldMap, firstIndex, lastIndex);
                }
            }

            wwb.write();
            wwb.close();

        }catch (Exception e){
            e.printStackTrace();
            //如果是ExcelException,则直接抛出
            if (e instanceof ExcelException){
                throw (ExcelException)e;

            //否则将其他异常包装成ExcelException再抛出
            }else {
                throw new ExcelException("导出Excel失败");
            }
        }

    }

    /**
     * @MethodName  : listToExcel
     * @Description : 导出Excel（可以导出到本地文件系统，也可以导出到浏览器，工作表大小为2003支持的最大值）
     * @param list      数据源
     * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
     * @param sheetName 工作表的名称
     * @param os       导出流
     * @throws ExcelException
     */
    public static <T> void listToExcel(
            List<T> list,
            LinkedHashMap<String, String> fieldMap,
            String sheetName,
            OutputStream os
    )throws ExcelException{
        listToExcel(list, fieldMap, sheetName, 65535, os);
    }

    /**
     * @MethodName  : listToExcel
     * @Description : 导出Excel（导出到浏览器，可以自定义工作表的大小）
     * @param list      数据源
     * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
     * @param sheetName 工作表的名称
     * @param sheetSize    每个工作表中记录的最大个数
     * @param response  使用response可以导出到浏览器
     * @throws ExcelException
     */
    public static <T> void listToExcel(
            List<T> list,
            LinkedHashMap<String, String> fieldMap,
            String sheetName,
            int sheetSize,
            HttpServletResponse response
    )throws ExcelException{
        //文件名默认设置为当前时间：年月日时分秒
        String fileName = DateFormatUtils.format(new Date(), "yyyyMMddhhmmss");
        //设置response头信息
        response.reset();
        response.setContentType("application/vnd.ms-excel");
        response.setHeader("Content-disposition","attachment;filename="+fileName+".xls");

        //创建工作簿并发送到浏览器
        try{
            OutputStream os = response.getOutputStream();
        }catch (Exception e){
            e.printStackTrace();

            //如果是ExcelException,则直接抛出
            if (e instanceof ExcelException){
                throw (ExcelException)e;
            }else {
                //否则将其他异常包装成ExcelException再抛出
                throw new ExcelException("导出excel失败");
            }
        }
    }

    /**
     * @MethodName  : listToExcel
     * @Description : 导出Excel（导出到浏览器，工作表的大小是2003支持的最大值）
     * @param list      数据源
     * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系
     * @param sheetName 工作表的名称
     * @param response  使用response可以导出到浏览器
     * @throws ExcelException
     */
    public static <T>  void   listToExcel (
            List<T> list ,
            LinkedHashMap<String,String> fieldMap,
            String sheetName,
            HttpServletResponse response
    ) throws ExcelException{

        listToExcel(list, fieldMap, sheetName, 65535, response);
    }

    /**
     * @MethodName : fillSheet
     * @Description : 向工作表中填充数据
     * @param sheet 工作表
     * @param list 数据源数据
     * @param fieldMap 中英文属性对照关系map
     * @param firstIndex 开始索引
     * @param lastIndex 结束索引
     * @param <E>
     * @throws Exception
     */
    private static <E> void fillSheet(
            WritableSheet sheet,
            List<E> list,
            LinkedHashMap<String, String> fieldMap,
            int firstIndex,
            int lastIndex
    )throws Exception{
        //定义存放英文字段名和中文字段名的数组
        int size = fieldMap.size();
        String[] enFields = new String[size];
        String[] cnFields = new String[size];

        //填充数组
        int count = 0;
        for (Map.Entry<String, String> entry:fieldMap.entrySet()){
            enFields[count] = entry.getKey();
            cnFields[count] = entry.getValue();
            count++;
        }

        //填充表头
        for (int i=0;i<cnFields.length;i++){
            Label label = new Label(i, 0 ,cnFields[i]);
            sheet.addCell(label);
        }

        //填充内容
        int rowNo = 1;
        for (int index=firstIndex;index<=lastIndex;index++){
            E item = list.get(index);
            for (int i=0;i<enFields.length;i++){
                Object objValue = getFieldValueByNameSequence(enFields[i], item);
                String fieldValue = objValue==null ? "" : objValue.toString();
                Label label = new Label(i, rowNo, fieldValue);
                sheet.addCell(label);
            }
            rowNo++;
        }
        //设置自动列宽
        setColumnAutoSize(sheet, 5);
    }

    /**
     * @MethodName  : getFieldValueByNameSequence
     * @Description :
     * 根据带路径或不带路径的属性名获取属性值
     * 即接受简单属性名，如userName等，又接受带路径的属性名，如student.department.name等
     *
     * @param fieldNameSequence  带路径的属性名或简单属性名
     * @param o 对象
     * @return  属性值
     * @throws Exception
     */
    private static Object getFieldValueByNameSequence(String fieldNameSequence, Object o) throws Exception{
        Object value = null;
        //将fieldNameSequence进行拆分
        String[] attributes = fieldNameSequence.split("\\.");
        if (attributes.length == 1){
            value = getFieldValueByName(fieldNameSequence, o);
        }else {
            //根据属性名获取属性对象
            Object fieldObj = getFieldValueByName(attributes[0], o);
            String subFieldNameSequence = fieldNameSequence.substring(fieldNameSequence.indexOf(".")+1);
            value = getFieldValueByNameSequence(subFieldNameSequence, fieldObj);
        }

        return value;
    }

    /**
     * @MethodName  : getFieldValueByName
     * @Description : 根据字段名获取字段值
     * @param fieldName 字段名
     * @param o 对象
     * @return  字段值
     */
    private static Object getFieldValueByName(String fieldName, Object o) throws Exception{
        Object value = null;
        Field field = getFieldByName(fieldName, o.getClass());

        if (field != null){
            field.setAccessible(true);
            value = field.get(o);
        }else {
            throw new ExcelException(o.getClass().getSimpleName() + "类不存在字段名" +fieldName);
        }

        return value;
    }

    /**
     * @MethodName  : getFieldByName
     * @Description : 根据字段名获取字段
     * @param fieldName 字段名
     * @param clazz 包含该字段的类
     * @return 字段
     */
    private static Field getFieldByName(String fieldName, Class<?> clazz){
         //拿到本类所有的字段
         Field[] selfFields = clazz.getDeclaredFields();

         //如果本类中存放该字段，则返回
         for (Field field : selfFields){
             if (field.getName().equals(fieldName)){
                 return field;
             }
         }

         //否则，查看父类中是否存在此字段，如果有则返回
         Class<?> superClazz = clazz.getSuperclass();
         if (superClazz != null && superClazz != Object.class){
             return getFieldByName(fieldName, superClazz);
         }

         //如果本来和父类都没有该字段，则返回null
         return null;
     }


    /**
     * @MethodName  : setColumnAutoSize
     * @Description : 设置工作表自动列宽和首行加粗
     * @param ws
     */
    private static void setColumnAutoSize(WritableSheet ws,int extraWith){
        //获取本列的最宽单元格的宽度
        for (int i=0;i<ws.getColumns();i++){
            int colWith = 0;
            for (int j=0;j<ws.getRows();j++){
                String content = ws.getCell(i,j).getContents().toString();
                int cellWith = content.length();
                if (colWith < cellWith){
                    colWith = cellWith;
                }
            }
            //设置单元格的宽度为最宽宽度+额外宽度
            ws.setColumnView(i, colWith+extraWith);
        }
    }

    /**
     * @Description 将Excel转化成实体对象List
     * @param is            要导入Excel的输入流
     * @param sheetName     导入的工作表名称
     * @param entityClass   List中对象的类型（Excel中的每一行都要转化为该类型的对象）
     * @param fieldMap      类的英文属性和Excel中的中文列名的对应关系 例：{id=编号}
     * @param uniqueFields  指定业务主键组合（即复合主键），这些列的组合不能重复
     * @return List
     * @throws ExcelException
     */
    public static<T> List<T> excelToList(
            InputStream is,
            String sheetName,
            Class<T> entityClass,
            LinkedHashMap<String, String> fieldMap,
            String[] uniqueFields
    )throws ExcelException{
        //定义要返回的list
        List<T> resultList = new ArrayList<T>();

        try {
            //根据excel数据源创建WorkBook
            Workbook wb = Workbook.getWorkbook(is);

            //获取工作表
            Sheet sheet = wb.getSheet(sheetName);

            //获取工作表的有效行数
            int realRows = 0 ;
            for (int i=0;i<sheet.getRows();i++){

                int nullCols = 0;
                for (int j=0;j<sheet.getColumns();j++){
                    Cell CurrentCell = sheet.getCell(j, i);
                    if (CurrentCell == null || "".equals(CurrentCell.getContents().toString())){
                        nullCols++;
                    }
                }

                if (nullCols == sheet.getColumns()){
                    break;
                }else {
                    realRows++;
                }
            }

            //如果Excel中没有任何数据则提示错误信息
            if (realRows <= 1){
                throw new ExcelException("Excel文件中没有任何数据");
            }

            Cell[] firstRow = sheet.getRow(0);
            String[] excelFieldNames = new String[firstRow.length];
            //获取Excel的列名
            for (int i=0;i<firstRow.length;i++){
                excelFieldNames[i] = firstRow[i].getContents().toString().trim();
            }
            //判断需要的字段在Excel中是否都存在
            boolean isExist = true;
            List<String> excelFieldList = Arrays.asList(excelFieldNames);
            for (String cnName : fieldMap.values()){
                if (!excelFieldList.contains(cnName)){
                    isExist = false;
                    break;
                }
            }

            //如果有列名不存在或不匹配，则抛出异常并提示错误
            if (!isExist){
                throw new ExcelException("Excel中缺少必要的字段，或字段名称有误");
            }

            //将列名和列号放入Map中，这样通过列名就可以拿到列号
            LinkedHashMap<String, Integer> colMap = new LinkedHashMap<String, Integer>();
            for (int i=0;i<excelFieldNames.length;i++){
                colMap.put(excelFieldNames[i], firstRow[i].getColumn());
            }

            //判断是否有重复行
            //1.获取uniqueFields指定的列
            Cell[][] uniqueCells = new Cell[uniqueFields.length][];
            for (int i=0;i<uniqueFields.length;i++){
                int col = colMap.get(uniqueFields[i]);
                uniqueCells[i] = sheet.getColumn(col);
            }
            //2.从指定列中寻找重复行
            for (int i=1;i<realRows;i++){
                int nullCols = 0;
                int length = uniqueFields.length;
                for (int j=0;j<length;j++){
                    Cell currentCell = uniqueCells[j][i];
                    String currentContent = currentCell.getContents().toString().trim();
                    Cell sameCell = sheet.findCell(currentContent,
                            currentCell.getColumn(),
                            currentCell.getRow()+1,
                            currentCell.getColumn(),
                            uniqueCells[j][realRows-1].getRow(),
                            true);
                    if (sameCell != null){
                        nullCols++;
                    }
                }
                //复合主键，意味着这些列的组合不能重复，
                // 只有当所有的列都有重复的时候，才被认为是有重复行
                if (nullCols == length){
                    throw new Exception("Excel中有重复行，请检查");
                }
            }

            //将sheet转换为list
            for (int i=1;i<realRows;i++){
                //新建要转换的对象
                T entity = entityClass.newInstance();

                //给对象中的字段赋值
                for (Map.Entry<String, String> entry : fieldMap.entrySet()){
                    //获取英文字段名
                    String enNormalName = entry.getKey();
                    //获取中文字段名
                    String cnNormalName = entry.getValue();
                    //根据中文字段名获取列号
                    int col = colMap.get(cnNormalName);

                    //获取当前单元格中的内容
                    String content = sheet.getCell(col, i).getContents().toString().trim();

                    //给对象赋值
                    setFieldValueByName(enNormalName, content, entity);
                }

                resultList.add(entity);
            }
        } catch (Exception e) {
            e.printStackTrace();
            //如果是ExcelException,则直接抛出
            if (e instanceof ExcelException){
                throw (ExcelException)e;
            }else {
                //否则将其包装成ExcelException,再将其抛出
                throw new ExcelException("导入ExceL失败");
            }
        }

        return resultList;
    }

    /**
     * @Description 根据字段名给对象的字段赋值
     * @param fieldName  字段名
     * @param fieldValue 字段值
     * @param o          对象
     */
    private static void setFieldValueByName(
            String fieldName,Object fieldValue,Object o)throws Exception{
        Field field = getFieldByName(fieldName, o.getClass());
        if (field != null){
            field.setAccessible(true);
            //获取字段类型
            Class<?> fieldType = field.getType();

            //根据字段类型给字段赋值
            if (String.class == fieldType){
                field.set(o, String.valueOf(fieldValue));
            } else if (Integer.TYPE == fieldType || Integer.class == fieldType){
                field.set(o, Integer.valueOf(fieldValue.toString()));
            } else if (Long.TYPE == fieldType || Long.class == fieldType){
                field.set(o, Long.valueOf(fieldValue.toString()));
            } else if (Float.TYPE == fieldType || Float.class == fieldType){
                field.set(o, Float.valueOf(fieldValue.toString()));
            } else if (Short.TYPE == fieldType || Short.class == fieldType){
                field.set(o, Short.valueOf(fieldValue.toString()));
            } else if (Double.TYPE == fieldType || Double.class == fieldType){
                field.set(o, Double.valueOf(fieldValue.toString()));
            } else if (Character.TYPE == fieldType){
                if ((fieldValue != null) && (fieldValue.toString().length() > 0)){
                    field.set(o, Character.valueOf(fieldValue.toString().charAt(0)));
                }
            } else if (Date.class == fieldType){
                field.set(o, DateUtils.parseDate(fieldValue.toString(),new String[]{"yyyy-MM-dd HH:mm:ss"}));
            } else {
                field.set(o, fieldValue);
            }
        } else {
            throw new ExcelException(o.getClass().getSimpleName() + "类不存在字段名" + fieldName);
        }
    }

    public static void main(String args[]) throws Exception {
        try {
            List<Student> students = new ArrayList<Student>();
            Student s1 = new Student();
            s1.setId(1);
            s1.setName("Tom");
            s1.setScore(78);
            students.add(s1);

            Student s2 = new Student();
            s2.setId(2);
            s2.setName("Hanks");
            s2.setScore(56);
            students.add(s2);

            Student s3 = new Student();
            s3.setId(3);
            s3.setName("jerry");
            s3.setScore(99);
            students.add(s3);

            LinkedHashMap<String, String> fieldMap = new LinkedHashMap<String, String>();
            fieldMap.put("id", "编号");
            fieldMap.put("name", "姓名");
            fieldMap.put("score", "分数");

            File osFile = new File("e:/test.xls");
            FileOutputStream fos = new FileOutputStream(osFile);
            listToExcel(students, fieldMap, "studentScore", fos);
            System.out.println("download success!");
        }catch (Exception e){
            throw new Exception("export error:" + e.getMessage());
        }
    }
}
